from calculator import Count
import unittest


class TestAdd(unittest.TestCase):
    def setUp(self):
        print('test add start')

    def tearDown(self):
        print('test add end')

    def test_add(self):
        j = Count(2, 3)
        self.assertEqual(j.add(), 5)

    def test_add2(self):
        j = Count(41, 76)
        self.assertEqual(j.add(), 117)


if __name__ == '__main__':
    unittest.main()
